#ifndef RKNNTOOL_H
#define RKNNTOOL_H
#include <opencv2/opencv.hpp>
#include "inc/rknn_api.h"
#include <sys/time.h>

typedef uint16_t fp16_t;
void dump_tensor_attr(rknn_tensor_attr *attr);
unsigned char *load_model(const char *filename, int *model_size);
int rknn_GetTopN(float *pfProb, float *pfMaxProb, uint32_t *pMaxClass, uint32_t outputCount, uint32_t topNum);
int64_t getCurrentTimeUs();
void save_image(const float* data, int width, int height, const std::string& filename);
void print_fp16_value(fp16_t *address);


#endif // RKNNTOOL_H
